from django import forms
from .models import CategoryModel, ProductModel 

class CategoryAddForm(forms.ModelForm):
    class Meta:
        model = CategoryModel
        fields = ("name",)

class ProductAddForm(forms.ModelForm):
    class Meta:
        model = ProductModel
        fields = ("name","spec", "category","total",)
